package own.stu.jobgib.playown.alg.leetcode.editor.cn;

//给你一个按 非递减顺序 排序的整数数组 nums，返回 每个数字的平方 组成的新数组，要求也按 非递减顺序 排序。 
//
// 
// 
//
// 
//
// 示例 1： 
//
// 
//输入：nums = [-4,-1,0,3,10]
//输出：[0,1,9,16,100]
//解释：平方后，数组变为 [16,1,0,9,100]
//排序后，数组变为 [0,1,9,16,100] 
//
// 示例 2： 
//
// 
//输入：nums = [-7,-3,2,3,11]
//输出：[4,9,9,49,121]
// 
//
// 
//
// 提示： 
//
// 
// 1 <= nums.length <= 104 
// -104 <= nums[i] <= 104 
// nums 已按 非递减顺序 排序 
// 
//
// 
//
// 进阶： 
//
// 
// 请你设计时间复杂度为 O(n) 的算法解决本问题 
// 
// Related Topics 数组 双指针 
// 👍 217 👎 0



public class SquaresOfASortedArray{
    public static void main(String[] args) {
        Solution solution = new SquaresOfASortedArray().new Solution();
        solution.sortedSquares(new int[]{-4,-1,0,3,10});
        solution.sortedSquares(new int[]{-7,-3,2,3,11});
        solution.sortedSquares(new int[]{-5,-3,-2,-1});
        solution.sortedSquares(new int[]{-10000,-9999,-7,-5,0,0,10000});
        solution.sortedSquares(new int[]{-4,-1,0,3,10});
    }
    //leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public int[] sortedSquares(int[] nums) {
        if(nums == null || nums.length == 0){
            return nums;
        }

        int len = nums.length;

        int m = -1;
        for(int i = 0; i <= len - 2; i ++){
            if(Math.abs(nums[i]) < Math.abs(nums[i + 1])){
                m = i;
                break;
            }
        }
        if(m == -1){
            m = len - 1;
        }

        int[] ans = new int[len];

        int idx = 0;
        int l = m, r = m + 1;
        while (0 <= l && r <= len - 1) {
            if (Math.abs(nums[l]) < Math.abs(nums[r])) {
                ans[idx++] = nums[l] * nums[l];
                l--;
            } else {
                ans[idx++] = nums[r] * nums[r];
                r++;
            }
        }

        while (0 <= l) {
            ans[idx++] = nums[l] * nums[l];
            l--;
        }
        while (r <= len - 1) {
            ans[idx++] = nums[r] * nums[r];
            r++;
        }

        return ans;
    }
}
//leetcode submit region end(Prohibit modification and deletion)

}
